export HCCL_WHITELIST_DISABLE=1
RANK_ID_START=0
WORLD_SIZE=8
for((RANK_ID=$RANK_ID_START;RANK_ID<$((WORLD_SIZE+RANK_ID_START));RANK_ID++));
do
    echo "Device ID: $RANK_ID"
    export LOCAL_RANK=$RANK_ID
    python -u run.py \
      --task_name anomaly_detection \
      --is_training 1 \
      --root_path ./dataset/PSM \
      --model_id PSM \
      --model TimesNet \
      --data PSM \
      --features M \
      --seq_len 100 \
      --pred_len 0 \
      --d_model 64 \
      --d_ff 64 \
      --e_layers 2 \
      --enc_in 25 \
      --c_out 25 \
      --top_k 3 \
      --anomaly_ratio 1 \
      --batch_size 128 \
      --train_epochs 2 \
      --world_size $WORLD_SIZE \
      --use_multi_gpu \
      --num_workers 1 &
done
wait